<!DOCTYPE html>
<html lang="en">
	<head>
		<title>three.js xr - paint</title>
		<meta charset="utf-8">
		<meta name="viewport" content="width=device-width, initial-scale=1.0, user-scalable=no">
		<link type="text/css" rel="stylesheet" href="main.css">
	</head>
	<body>

		<div id="info">
			<a href="https://threejs.org" target="_blank" rel="noopener">three.js</a> xr - paint
		</div>

		<script type="importmap">
			{
				"imports": {
					"three": "../build/three.module.js",
					"three/addons/": "./jsm/"
				}
			}
		</script>

		<script type="module">

			import * as THREE from 'three';
			import { OrbitControls } from 'three/addons/controls/OrbitControls.js';
			import { TubePainter } from 'three/addons/misc/TubePainter.js';
			import { XRButton } from 'three/addons/webxr/XRButton.js';

			let camera, scene, renderer;
			let controller1, controller2;

			const cursor = new THREE.Vector3();
			const color = new THREE.Color();

			let controls;

			init();

			function init() {

				const container = document.createElement( 'div' );
				document.body.appendChild( container );

				scene = new THREE.Scene();
				scene.background = new THREE.Color( 0x222222 );

				camera = new THREE.PerspectiveCamera( 50, window.innerWidth / window.innerHeight, 0.01, 50 );
				camera.position.set( 0, 1.6, 3 );

				controls = new OrbitControls( camera, container );
				controls.target.set( 0, 1.6, 0 );
				controls.update();

				const grid = new THREE.GridHelper( 4, 1, 0x111111, 0x111111 );
				scene.add( grid );

				scene.add( new THREE.HemisphereLight( 0x888877, 0x777788, 3 ) );

				const light = new THREE.DirectionalLight( 0xffffff, 1.5 );
				light.position.set( 0, 4, 0 );
				scene.add( light );

				//

				const painter1 = new TubePainter();
				scene.add( painter1.mesh );

				const painter2 = new TubePainter();
				scene.add( painter2.mesh );

				//

				renderer = new THREE.WebGLRenderer( { antialias: true } );
				renderer.setPixelRatio( window.devicePixelRatio );
				renderer.setSize( window.innerWidth, window.innerHeight );
				renderer.setAnimationLoop( animate );
				renderer.xr.enabled = true;
				container.appendChild( renderer.domElement );

				document.body.appendChild( XRButton.createButton( renderer ) );

				// controllers

				function onSelectStart() {

					this.updateMatrixWorld( true );

					const pivot = this.getObjectByName( 'pivot' );
					cursor.setFromMatrixPosition( pivot.matrixWorld );

					const painter = this.userData.painter;
					painter.moveTo( cursor );

					this.userData.isSelecting = true;

				}

				function onSelectEnd() {

					this.userData.isSelecting = false;

				}

				function onSqueezeStart() {

					this.userData.isSqueezing = true;
					this.userData.positionAtSqueezeStart = this.position.y;
			
					const pivot = this.getObjectByName( 'pivot' );
					this.userData.scaleAtSqueezeStart = pivot.scale.x;

				}

				function onSqueezeEnd() {

					this.userData.isSqueezing = false;

				}

				controller1 = renderer.xr.getController( 0 );
				controller1.addEventListener( 'selectstart', onSelectStart );
				controller1.addEventListener( 'selectend', onSelectEnd );
				controller1.addEventListener( 'squeezestart', onSqueezeStart );
				controller1.addEventListener( 'squeezeend', onSqueezeEnd );
				controller1.userData.painter = painter1;
				scene.add( controller1 );

				controller2 = renderer.xr.getController( 1 );
				controller2.addEventListener( 'selectstart', onSelectStart );
				controller2.addEventListener( 'selectend', onSelectEnd );
				controller2.addEventListener( 'squeezestart', onSqueezeStart );
				controller2.addEventListener( 'squeezeend', onSqueezeEnd );
				controller2.userData.painter = painter2;
				scene.add( controller2 );

				//

				const pivot = new THREE.Mesh( new THREE.IcosahedronGeometry( 0.01, 3 ) );
				pivot.name = 'pivot';
				pivot.position.z = - 0.05;

				const group = new THREE.Group();
				group.add( pivot );

				controller1.add( group.clone() );
				controller2.add( group.clone() );

				//

				window.addEventListener( 'resize', onWindowResize );

			}

			function onWindowResize() {

				camera.aspect = window.innerWidth / window.innerHeight;
				camera.updateProjectionMatrix();

				renderer.setSize( window.innerWidth, window.innerHeight );

			}

			//

			function handleController( controller ) {

				controller.updateMatrixWorld( true );

				const userData = controller.userData;
				const painter = userData.painter;

				const pivot = controller.getObjectByName( 'pivot' );

				if ( userData.isSqueezing === true ) {

					const delta = ( controller.position.y - userData.positionAtSqueezeStart ) * 5;
					const scale = Math.max( 0.1, userData.scaleAtSqueezeStart + delta );

					pivot.scale.setScalar( scale );
					painter.setSize( scale );

				}

				cursor.setFromMatrixPosition( pivot.matrixWorld );

				if ( userData.isSelecting === true ) {

					const hue = ( performance.now() * 0.001 ) % 1;
					color.setHSL( hue, 1.0, 0.5 );
					painter.setColor( color );

					painter.lineTo( cursor );
					painter.update();

				}

			}

			function animate() {

				handleController( controller1 );
				handleController( controller2 );

				renderer.render( scene, camera );

			}

		</script>
	</body>
</html>
